import pandas as pd
from openpyxl import Workbook
from openpyxl.utils.dataframe import dataframe_to_rows

zcfz_file = '汽车零部件行业资产负债表.xlsx'
lrb_file = '汽车零部件行业利润表.xlsx'
xjll_file = '汽车零部件行业现金流量表.xlsx'

# 读取三个 Excel 文件
df_zcfz = pd.read_excel(zcfz_file)
df_lrb = pd.read_excel(lrb_file)
df_xjll = pd.read_excel(xjll_file)

# 创建一个新的 Excel 工作簿
wb = Workbook()

# 添加三个工作表
ws_zcfz = wb.create_sheet(title='汽车零部件行业资产负债表2018-2023')
ws_lrb = wb.create_sheet(title='汽车零部件行业利润表2018-2023')
ws_xjll = wb.create_sheet(title='汽车零部件行业现金流量表2018-2023')


# 将数据写入工作表
def write_dataframe_to_sheet(ws, df):
    for r in dataframe_to_rows(df, index=False, header=True):
        ws.append(r)


write_dataframe_to_sheet(ws_zcfz, df_zcfz)
write_dataframe_to_sheet(ws_lrb, df_lrb)
write_dataframe_to_sheet(ws_xjll, df_xjll)

# 删除默认的工作表
del wb['Sheet']

# 保存新的 Excel 文件
output_file = '汽车零部件行业财务报表2018-2023.xlsx'
wb.save(output_file)

print(f"数据已保存到 {output_file}")
